import importlib

terrain_registry = dict(
    NavTrack= "legged_gym.utils.nav_terrain.nav_track:NavTrack",
)

def get_nav_track(terrain_cls):
    entry_point = terrain_registry[terrain_cls]
    module, class_name = entry_point.rsplit(":", 1)
    module = importlib.import_module(module)
    return getattr(module, class_name)
